package com.baihang.project.ds.webSocket;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
@Configuration
@EnableWebSocket
public class WebSocketConfig  implements WebSocketConfigurer {

	@Override
	public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
		registry.addHandler(dsHandler(), "/Handler").addInterceptors(new DsHandshakeInterceptor()).setAllowedOrigins("http://localhost");

	}
	
	@Bean
    public WebSocketHandler dsHandler() {
        return new DsWebSocketHandler();
    }

}
